from pydantic import BaseModel
from typing import List, Dict
from openai import Agent

class UseCaseDSL(BaseModel):
    name: str
    actors: List[str]
    usecases: List[Dict[str, any]]

agent_use_case = Agent(
    name="UseCaseGenerator",
    instructions=(
        "根据需求描述生成用例图DSL，包括Actor、Use Cases以及它们的关系（include/extend）。"
    ),
    output_type=UseCaseDSL,
    model="gpt-4o"
)
